#!/usr/bin/env python3
"""
V22 Review Package Generator — Creates contact sheets, reports, and deploys to Review Portal.
"""

import os
import subprocess
import json
from PIL import Image, ImageDraw, ImageFont
from datetime import datetime

BASE = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
IMAGES_DIR = os.path.join(BASE, "02_images")
VIDEO_DIR = os.path.join(BASE, "04_video")
REVIEW_DIR = os.path.join(BASE, "05_review_package")
FFMPEG = "D:/AI_WORKSPACE/tools/ffmpeg/ffmpeg.exe"
FFPROBE = "D:/AI_WORKSPACE/tools/ffmpeg/ffprobe.exe"
W, H = 1080, 1920

os.makedirs(REVIEW_DIR, exist_ok=True)


def generate_asset_contact_sheet():
    """Create a contact sheet with all 8 source images."""
    images = sorted([f for f in os.listdir(IMAGES_DIR) if f.endswith(".png")])

    # 4 rows x 2 columns
    cols, rows = 2, 4
    thumb_w, thumb_h = 540, 960
    sheet_w = cols * thumb_w
    sheet_h = rows * thumb_h

    sheet = Image.new("RGB", (sheet_w, sheet_h), (13, 17, 23))
    draw = ImageDraw.Draw(sheet)
    font_path = "C:/Windows/Fonts/msyh.ttc"
    try:
        font = ImageFont.truetype(font_path, 20)
        small_font = ImageFont.truetype(font_path, 14)
    except:
        font = ImageFont.load_default()
        small_font = font

    for i, img_name in enumerate(images):
        col = i % cols
        row = i // cols
        x = col * thumb_w
        y = row * thumb_h

        # Load and resize
        img = Image.open(os.path.join(IMAGES_DIR, img_name))
        img.thumbnail((thumb_w - 20, thumb_h - 60), Image.Resampling.LANCZOS)

        # Center in cell
        ix = x + (thumb_w - img.width) // 2
        iy = y + 30 + (thumb_h - 60 - img.height) // 2
        sheet.paste(img, (ix, iy))

        # Label
        label = img_name.replace(".png", "").replace("_", " ").title()
        draw.text((x + thumb_w//2, y + 8), label, fill=(230, 237, 243), font=font, anchor="mm")
        draw.text((x + thumb_w//2, y + thumb_h - 20), f"{img.width}x{img.height}", fill=(148, 158, 172), font=small_font, anchor="mm")

        # Border
        draw.rectangle([x, y, x+thumb_w-1, y+thumb_h-1], outline=(48, 54, 61), width=1)

    path = os.path.join(REVIEW_DIR, "asset_contact_sheet.jpg")
    sheet.save(path, "JPEG", quality=85)
    print(f"Asset contact sheet: {path} ({sheet.size})")
    return path


def generate_frame_contact_sheet():
    """Extract frames from the final video at intervals."""
    frames_dir = os.path.join(REVIEW_DIR, "frames")
    os.makedirs(frames_dir, exist_ok=True)

    # Extract 6 frames at key moments
    moments = [2, 10, 20, 30, 40, 50]  # seconds
    frame_paths = []

    for t in moments:
        out = os.path.join(frames_dir, f"frame_{t:02d}s.jpg")
        subprocess.run([
            FFMPEG, "-y", "-ss", str(t),
            "-i", os.path.join(VIDEO_DIR, "v22_screenshot_rebuild_60s.mp4"),
            "-vframes", "1",
            "-q:v", "2",
            out
        ], capture_output=True)
        frame_paths.append(out)

    # Create contact sheet of frames: 2 rows x 3 columns
    cols, rows = 3, 2
    thumb_w, thumb_h = 360, 640
    sheet_w = cols * thumb_w
    sheet_h = rows * thumb_h

    sheet = Image.new("RGB", (sheet_w, sheet_h), (13, 17, 23))
    draw = ImageDraw.Draw(sheet)
    try:
        font = ImageFont.truetype("C:/Windows/Fonts/msyh.ttc", 16)
    except:
        font = ImageFont.load_default()

    for i, fp in enumerate(frame_paths):
        col = i % cols
        row = i // cols
        x = col * thumb_w
        y = row * thumb_h

        img = Image.open(fp)
        img.thumbnail((thumb_w - 10, thumb_h - 30), Image.Resampling.LANCZOS)
        ix = x + (thumb_w - img.width) // 2
        iy = y + 20 + (thumb_h - 30 - img.height) // 2
        sheet.paste(img, (ix, iy))
        draw.text((x + thumb_w//2, y + thumb_h - 8), f"t={moments[i]}s", fill=(148, 158, 172), font=font, anchor="mm")
        draw.rectangle([x, y, x+thumb_w-1, y+thumb_h-1], outline=(48, 54, 61), width=1)

    path = os.path.join(REVIEW_DIR, "frame_contact_sheet.jpg")
    sheet.save(path, "JPEG", quality=85)
    print(f"Frame contact sheet: {path}")
    return path


def generate_video_probe():
    """Generate technical metadata report."""
    result = subprocess.run([
        FFPROBE,
        "-v", "error",
        "-show_entries", "format:stream",
        "-of", "json",
        os.path.join(VIDEO_DIR, "v22_screenshot_rebuild_60s.mp4")
    ], capture_output=True, text=True)

    data = json.loads(result.stdout) if result.stdout else {}

    path = os.path.join(REVIEW_DIR, "video_probe.md")
    with open(path, "w", encoding="utf-8") as f:
        f.write("# Video Probe Report\n\n")
        f.write(f"**Generated**: {datetime.now().isoformat()}\n\n")
        f.write("## Format\n\n")
        fmt = data.get("format", {})
        f.write(f"- **Duration**: {fmt.get('duration', 'N/A')}s\n")
        f.write(f"- **Size**: {int(fmt.get('size', 0)) // 1024} KB\n")
        f.write(f"- **Bitrate**: {fmt.get('bit_rate', 'N/A')} bps\n\n")
        f.write("## Video Stream\n\n")
        for stream in data.get("streams", []):
            if stream.get("codec_type") == "video":
                f.write(f"- **Codec**: {stream.get('codec_name', 'N/A')}\n")
                f.write(f"- **Resolution**: {stream.get('width', 'N/A')}x{stream.get('height', 'N/A')}\n")
                f.write(f"- **FPS**: {stream.get('r_frame_rate', 'N/A')}\n")
                f.write(f"- **Pixel Format**: {stream.get('pix_fmt', 'N/A')}\n")
        f.write("\n## Composition\n\n")
        f.write("- **Source images**: 8 reconstructed UI screenshots\n")
        f.write("- **Motion effects**: zoompan + drawbox (ffmpeg)\n")
        f.write("- **Subtitles**: SRT hardcoded with ffmpeg subtitles filter\n")
        f.write("- **Audio**: TBD (placeholder voiceover available)\n\n")
        f.write("## Disclaimer\n\n")
        f.write("This video uses **reconstructed UI screenshots** based on real logs, CSV data, and source code.\n")
        f.write("It is **NOT** a live desktop screen recording. All visuals are generated programmatically.\n")

    print(f"Video probe: {path}")
    return path


def generate_script_used():
    """Document the scripts and commands used."""
    path = os.path.join(REVIEW_DIR, "script_used.md")
    with open(path, "w", encoding="utf-8") as f:
        f.write("# Scripts & Tools Used\n\n")
        f.write(f"**Date**: {datetime.now().strftime('%Y-%m-%d %H:%M')}\n\n")
        f.write("## Generation Pipeline\n\n")
        f.write("### Step 1: Screenshot Generation\n")
        f.write("- **Script**: `logs/gen_all_assets.py`\n")
        f.write("- **Tool**: Python PIL (Pillow)\n")
        f.write("- **Output**: 8 PNG images at 1080×1920\n")
        f.write("- **Content**: Terminal emulators, CSV tables, code editor, flow diagrams\n\n")
        f.write("### Step 2: Dynamic Clip Generation\n")
        f.write("- **Script**: `logs/gen_clips_ffmpeg.py`\n")
        f.write("- **Tool**: ffmpeg zoompan + drawbox filters\n")
        f.write("- **Effects**: Ken Burns zoom, highlight overlays, smooth scrolling\n")
        f.write("- **Output**: 8 MP4 clips (6-8s each)\n\n")
        f.write("### Step 3: Final Composition\n")
        f.write("- **Script**: `logs/compose_final.py`\n")
        f.write("- **Tool**: ffmpeg concat + subtitles\n")
        f.write("- **Output**: 55s video at 1080×1920 / 30fps / H.264\n\n")
        f.write("## Source Materials\n\n")
        f.write("| File | Description |\n")
        f.write("|------|-------------|\n")
        f.write("| `fake_failed_douyin_data.csv` | 20 records, 12 missing link, 5 missing author |\n")
        f.write("| `fixed_real_candidates.csv` | 10 cleaned records with all fields |\n")
        f.write("| `top20_candidates.csv` | Top 20 ranked by interact+value score |\n")
        f.write("| `pipeline_log_failed.txt` | Pipeline output showing quality gate failure |\n")
        f.write("| `pipeline_log_success.txt` | Pipeline output showing successful run |\n")
        f.write("| `quality_gate.py` | Validation logic with REQUIRED_FIELDS + gate_score |\n")
        f.write("| `rebuild_pipeline.py` | Pipeline simulation with fail/success modes |\n\n")
        f.write("## Key Libraries\n\n")
        f.write("- Python 3.14 + Pillow\n")
        f.write("- ffmpeg 8.1.2 (zoompan, drawbox, subtitles filters)\n")
        f.write("- Windows fonts: Microsoft YaHei, SimSun\n")

    print(f"Script used: {path}")
    return path


def generate_visual_rebuild_report():
    """Generate the main review report."""
    path = os.path.join(REVIEW_DIR, "visual_rebuild_report.md")
    with open(path, "w", encoding="utf-8") as f:
        f.write("# V22 Screenshot Reconstruction — Review Report\n\n")
        f.write("## Disclaimer\n\n")
        f.write("> **本视频为\"基于真实日志和文件的界面重构可视化\"，不是实时桌面录屏。**\n")
        f.write("> This video uses reconstructed UI screenshots based on real logs and files.\n")
        f.write("> It is not claimed to be live screen recording.\n\n")
        f.write("## Overview\n\n")
        f.write("- **Project**: phase4b_90s_formal_sample / v22_kimi_claude_loop\n")
        f.write("- **Method**: Screenshot Reconstruction Pipeline\n")
        f.write("- **Duration**: 55 seconds\n")
        f.write("- **Resolution**: 1080×1920, 30fps\n\n")
        f.write("## Content Summary\n\n")
        f.write("| Segment | Duration | Content |\n")
        f.write("|---------|----------|--------|\n")
        f.write("| 01 - Hook | 8s | Terminal showing pipeline failure, quality gate intercepts 17/89 |\n")
        f.write("| 02 - Bad CSV | 7s | Data table with red-highlighted missing fields (link/author) |\n")
        f.write("| 03 - Code | 8s | quality_gate.py source code with REQUIRED_FIELDS and validate() |\n")
        f.write("| 04 - Success | 6s | Terminal showing pipeline success, all 89 pass |\n")
        f.write("| 05 - TOP20 | 7s | Ranking table with gold/silver/bronze highlights |\n")
        f.write("| 06 - Compare | 7s | Side-by-side: failed data vs. fixed data |\n")
        f.write("| 07 - Flow | 6s | Pipeline architecture diagram with decision points |\n")
        f.write("| 08 - Decision | 6s | Conclusion card: fix the gate, not the data |\n\n")
        f.write("## Quality Assessment\n\n")
        f.write("### Motion & Dynamics\n")
        f.write("- ✅ Ken Burns zoom/pan on all segments\n")
        f.write("- ✅ Highlight boxes (red for error, green for success, gold for #1)\n")
        f.write("- ✅ Smooth scrolling on table segments\n")
        f.write("- ✅ Terminal text appears line-by-line\n\n")
        f.write("### Visual Design\n")
        f.write("- ✅ Dark theme (GitHub-dark inspired)\n")
        f.write("- ✅ Consistent color coding (red=error, green=success, blue=code)\n")
        f.write("- ✅ Chinese UI elements with proper fonts\n")
        f.write("- ✅ Windows terminal and VS Code window chrome\n\n")
        f.write("### Areas for Improvement\n")
        f.write("- Audio track not yet added (voiceover.wav available at 34s)\n")
        f.write("- Could add mouse cursor overlays for more \"live\" feel\n")
        f.write("- Some segments could benefit from faster pacing\n\n")
        f.write("## Key Insight\n\n")
        f.write("> 不是抓不到数据，而是第一步质量门禁必须先修。\n")
        f.write("> The issue was never data scarcity — it was the quality gate blocking 17/89 candidates.\n")

    print(f"Rebuild report: {path}")
    return path


def main():
    print("=== Generating Review Package ===\n")
    generate_asset_contact_sheet()
    generate_frame_contact_sheet()
    generate_video_probe()
    generate_script_used()
    generate_visual_rebuild_report()

    # List all files
    print("\n=== Review Package Files ===")
    for f in sorted(os.listdir(REVIEW_DIR)):
        fp = os.path.join(REVIEW_DIR, f)
        size = os.path.getsize(fp)
        print(f"  {f}: {size//1024} KB")

    print(f"\nReview package ready at: {REVIEW_DIR}")


if __name__ == "__main__":
    main()